How to write a for loop in r