Refactor part_shift_dynamic.py