Updated documentation to be in line with changes.