Updated doc's to match code