Update docs to reflect latest state of the code.