fixed some formatting in the code