Add tests for task_1