Minor fix and documentation