Fixed output for week 3 question 6