Just some minor pep8 cleanup

We are using your library in our company and after peering in the code, we decided to contribute to you back by cleaning up the code a bit.