Snake Game in Python code