small changes in main