Fixed commenting for main function