Newton Fractals in the complex plane (python)