Learning from Trees: Two-Stage Enhancements