modify the diagram of binary tree in Main class