Some updates to tasks.py