Python Segment Tree Code