update hw4

almost there