Initial version for questions of quiz 2
Update code for question 4