Use absolute import for python2 compatibility
python version check for iteration removed