Lint fix

Added docstrings and to spaces everywhere.