Style improvements
Minor fix