Add some Neural Network functionality, still sloppy