Add support for 3D refinement