neural network python code example