Add Stokes axis at specific positon

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