R programming assignment - add comments