small play around with pygame