Add to Python notebook