Small code formatting updates