Wrote tests for Task2