Update main.R

Updated something