hw4 completed
minor changes to hw4