Basic Mandelbrot set in Python