Updated code for R Programming Assignment 2