changed if/else to switch statement because I forgot go had switch.