updated hw7 code