I used the relations environment to generate all legal moves.
TODO:
- comment more thoroughly
- make function map to pass to MCTS