Update cslm.py