Minor changes
Minor fix