Add python notebook support