minor changes to task-9