Added ability to weigh classes for SVM