Create segment_tree.py