Update docs too