Update for tasks.py