Modified SVM training script
Added prediction and training stuff