Playing around with pygame