solved problem 1 in r