Fix up code formatting