update main.R