add Hw3 code