For loop homework