Add tree structure to model, add some UI components