modify kmt.ipynb