Tried using switch statement instead, no luck

also setup a variable for color