Sample Neural Network Code Python