Minor Changes. Works with image segmentation very well