hw4 small mistake