Added a second argument to the main method