Minor python tweaks
There, cleaner.