Small code formatting update