modify geting the correct answer