Newton method in python