drop_N_columns tweaked with np.sum