Added code to show correct answer