Update env.py