Update task_3a.m