Add code for hw1