Mandelbrot Set Function In Python