Update main.R

layout