Linting fixes + better documentation