Add handling for volumetric models