Add drop while iterator problem