Add pygame stuff
Cleanup.