Small changes to hw4