Update docs and cleanup in example