HW3: 3.1 is done. All test cases work