Improve listening for notebook change