Minor tweaks to python files