Add bounding box functionality