Updates to tree rendering for v0.4.0