hw4 p1 updated