Add Stokes axis at specific position

Previously it was always appending to the end. This fixes one of the issues from #171 , where clean would crash from the outputted mask.