Cleanup
documentation
pep8