Add loop function
Test