adjust the API of KL methods for the notebook

@karlnapf 
please take a look at this