update formatting of code