Add returns to task_04