Add hw7 test code