Small code formatting changes