week 5 for loop change