HW 3 Part C - add tests