minor changes to .py