Added logic for showing correct answer type