Fixed Save bug in prediction engine