Update hw7

Add second part from hw6