MCTS refactoring