PEP8 in mass_dist