Updated the code-format a bit.