improved python notebook