A very simple neural network in python