Update python notebook!