Update mandelbrot.py