add codes for hw1