Update Assn_2.R