Edits to get scene segmentation going