Extend step4 example to 3D
Indent and comment updates