Yeah I'd use a switch statement only if there are > 2 options, could you adapt?