Removed custom TreeModel (we can probably work with DefaultTreeModel
instead)