1. Modify notebooks to adopt to the new API