Improvements to scheduling and train AI