updated imports and local path in task.py