Newton Method Python Code