Adjust code structure

I adjust the tree structure.
I hope you code can keep this clean structure.