Fix bad formatting of code