Got neural net up and running. Still needs some fine tuning