A simple mandelbrot script in python