minor tweaks to .py