Add docs, fix style