Add loop code