Add tests for task_6
Add task_6