Update srt_shifter.py