R programming assignment 2 revise