Fix imports and lower iterations