Made NN work better, but needs more training data.