Add tests for task_3