Fix a small bug in the Python script: prevent it from finding trees if it's the only tree under the same tree_id