Minor updates to python notebook modules

[RamiyapriyaS-NIH]

Summary

Minor updates to python notebooks to ensure code runs without errors.

Checklist

• [X ] My code follows the code style of this project.
• I have performed a self-review of my code.
• I have commented my code, particularly in hard-to-understand areas.
• I have made corresponding changes to the documentation.
• My changes generate no new warnings or errors.
• I have added tests that prove my fix is effective or that my feature works.
• New and existing unit tests pass locally with my changes.