Adds support for TaggedTrees with example.

Adds braces to if.
Removes unnecessary null check.