Python For Loop Change Iterator Code