Editted R programming assignment 2