Add code from hw1