Added a flag for axisymmetry problem