Still trying to create a correct model tree