Plot mandelbrot set using Pygame and NumPy