Remove doctests and clean up

* Some PEP8 here and there.

Commit closes #41.