Update shift.py