Upgrade `kshramt.py`