Separate class tree creation logic from tree view.

This will allow us to have multiple implementations of tree creation logic.