Treemap cleanup

Documented usage and also removed an extra line of code.