Update docs and add integration test