Basic [1 hidden layer] neural network on Python