adding correct answer logic