improve notebook demo