* Removed suspicious second prediction from training func