fixed homework 1 question 1 - another problem