Basic Neural Network Code In Python