Docs

Add docs build and refactor