Wrote some tests for Task 2