Cleaning up a bit in segment.py