small change to hw4