Neural Network Code works