Update main_code.R