Small code formatting fixes