Refactored separate environment classes back into main.py