Update docs as well