Small update to PyGame example.