Update OliviaLee_HW2_code.R