A worm game in pygame