Update assign1_rcode.R