also make it return the run 1 mask, as that is useful