small change in main.R