Update snake.py