fixing bug with correct answers