Actually wrote the switch statement correctly