Improve formatting

Follow the coding style used in the rest of the script.