Update formatting of the code