Add notebook utils