Added option to output the MCTS tree