Add shift support for this two loop setup