Test and train files complete. Fixing up neural net to interface with test/train