Python Code Enhancement & Integration

Python dev required to alter code in a Jupyter Viewer project. The first code works flawlessly. Have sample code to add the next feature in