need help with a switch

yup, that should work.