PEP8 + docstrings

@stefanv @ahojnnes et al. This is a quick fix.