Python Simple Snake Game Code