Python Snake Diagram