Update run.py with new tasks