minor updates to python script