Fixed the anwser for question 2