Add task_5 tests