A neural network in python using numpy