hw2 code