Adding set correct answer functionality