More code, made some high level decisions not to have a neural network class