Draw a Mandelbrot set with Python