Add docs and fix reference