Add shift function
Add missing parameter