Update csi.py