Modified for SGDClassifier

Achieved a 51% score with the modifications