update the KL notebook

fix some issues of the KL notebook
@karlnapf 
see this
http://nbviewer.ipython.org/gist/yorkerlin/135dba2625ac7e004d97