Update mandelbrot.py

Fix for Python 3.x