Python turtle code for snake game