Add small additions to Python notebook