Added switch in checkAnswer