The main part of environment is located in file "hexenv.py", other necessary functions are located in corresponding files. File "learnhex.py" is responsible for calling stable baselines 3 functions which allow agent to learn how to deal with environment.