Update docs due to api changes