minor changes in main