fix style issues
pep8
improve coverage