Wrote tests for the Task1 classes