Create mandelbrot.py