Wrote more tests for Task 2