Faffing around with pygame