updated python notebook