Updated docstrings for main env and simulator.py