Update wally segmentize.py