PMML: predict_proba functionality

Hi! Is it possible to add such a functionality to save_model method when saving in PMML format? Fow now it returns only final class labels