Add task_18
Add tests for task_18