Adding test for looping