Update main.R

The best model