Add show tree feature

New: ability to toggle the drawing of the underlying tree structure used
for calculating forces.
Change: simplify some code.