Minor fix
Updated docs