Adding env.py for the main env