Some minor cosmetic changes

Hi Schlomo,

I removed one unused import and added and removed some white space as it was suggested by flake8.

kind regards,
Michael