Update shift_assign.py