Update tasks.py to account for new structure.