Updated to train and pickle nn
Removed stupid commented code
Pickled neural net