WIP: Make trees and forests support string labels