fix formatting of code