Added vis for segments for demo. Other changes.