Assignment 2 for R programming course