Updated documentation to match code