Small game done in python with pygame