Update ch2hw2.R