Making the TreePair test script more user-friendly