Example fixed

Fixed a few small things in the example code. Nothing related to actual
tree stuff.