Minor fixes to python notebooks