Added segmentation + automatic box