Made some change to mandelbrot.py

Still needs lots of work