task1/index.py works with toy example