Add tests for task_2, modify task_2, add test script