Update model class to return bounding box for object class instead of
only center.