some test code to use PyGame