Added code for quiz