work on true 2D runs by setting nx=1