modification in main method