image segmentation visualisation added