Update formatting of code