fix pep8
Add docstings