Fix formatting of code