Small code formatting