homework #3 fix