fix half of issues in the KL notebook
Merge pull request #2605 from yorkerlin/develop

fix half of issues in the KL notebook