Adding env.py for the main env
Added Main environment class and main function