Add hw7 code