homework 3 . fix