Fix formatting in main