Update docs to match code. Refs #2