Added two switch statements to Play::handle_input