Update main.R file