Fixed answer for question #4