Fix up code styling