clean-up of test function to get trees