A little snake game in python with pygame