Pset 6 Q1 in R