Vectorising for loops to save time in R