How can I store an object in a research notebook and load it in the main.py? Storing in notebook works, but reading in main.py fails.