hw1 code