Fixed formatting in code