some more updates to hw4