Improved MCTS (WIP)