minor changes in main method