simple port to pygame

Made a simple port of the demo to pygame, to see any potential
performance differences. On my computer (2008 MacBook Pro) it doubled
the frame rate!