Fixed answer for question 4