Update task_3d.m