Fix up the code style