Add hooks to handle the tree data format
Add convenience function to load in some test data