Extended data_tree test. Should probably refactor out some of data_tree though