Changed SVM to NN