moved from script to Notebook