add correct answer functionality