Added some parameters in the main method