some more modern python usage

for review