cleanup (DOC, pep8 etc.)