Misc tweaks

- PEP 257
- Unbreak tests on Python 3
- setup.py clean and improvements