Experimenting with pygame