update the KL notebook

fix some issues of the KL notebook