Exercise 2 + bonus in for loops working