In Mafia, changing the amount of charges an ability during Setup has requires a separate check-box for each number. My suggestion is to replace the check-boxes with a Slider.
If the check-boxes are replaced with Sliders, most roles can be given two extra options during setup, allowing for additional customization. In these slots, roles like the Janitor can be given optional settings such as "Detection Immunity" while other roles like the Vigilante could be given the "Pierce Night Immunity" and "Immune at night" abilities.
Basically, Sliders should replace check-boxes because check-boxes take up too much space.