playing around with pygame