Small formatting of code