Add tests for task_12