Add Custom Iterator code