Minor fix and documentation update