Make the SimpleTree more voluminous.

Also separate the SimpleTree code to functions and improve visualisation
by playing with the parameters. Remove junk comments.