python coding for snake game