Update main.R

Comments of the script have been updated