The function gen_data takes the count of malicious+benign domains and adds this number of labels to the malicious labels.