Fractal in python, just for fun