Add iterator tests