Improved models for predicting solar flares