modified python notebook