Add task_6
Add tests for task_6