How to implement neural network in python