Update task_1
Add tests for task_1