Did some minor cleanup of displaying the correct answer for the question