Python Mandelbrot set generator