Python Code For Neural Net