add HW1 source code