Update assign2_method1.R