Looking for an easier way to do a switch