Fixed formatting of code