diff --git a/README.md b/README.md
index 99d3440..0f923dc 100644
--- a/README.md
+++ b/README.md
@@ -13,6 +13,10 @@ Builds on
 inspired by
 [LearningStrategies.jl](https://github.com/JuliaML/LearningStrategies.jl).
 
+The foundation for MLJ's
+[`IteratedModel`](https://juliaai.github.io/MLJ.jl/stable/controlling_iterative_models/)
+(provided by [MLJIteration.jl](https://github.com/JuliaAI/MLJIteration.jl)).
+
 Other related software:
 [DynamicIterators.jl](https://github.com/mschauer/DynamicIterators.jl).
 
diff --git a/docs/src/index.md b/docs/src/index.md
index 3dd57b9..9d06a09 100644
--- a/docs/src/index.md
+++ b/docs/src/index.md
@@ -8,6 +8,10 @@ Builds on
 inspired by
 [LearningStrategies.jl](https://github.com/JuliaML/LearningStrategies.jl).
 
+The foundation for MLJ's
+[`IteratedModel`](https://juliaai.github.io/MLJ.jl/stable/controlling_iterative_models/)
+(provided by [MLJIteration.jl](https://github.com/JuliaAI/MLJIteration.jl)).
+
 Other related software:
 [DynamicIterators.jl](https://github.com/mschauer/DynamicIterators.jl).
 
diff --git a/src/train.jl b/src/train.jl
index d66d79f..8e2d0b2 100644
--- a/src/train.jl
+++ b/src/train.jl
@@ -98,9 +98,8 @@ julia> traces[2].loss
 ```
 
 Alternatively, one may specify `traces=v`, where `v` is the vector of traces returned by a
-previous call to `train!` with the same control types. In that `v` is extended, and
-matches the vector returned by the second call to `train!`. Continuing the above
-example:
+previous call to `train!` with the same control types. In that case `v` is extended, and
+matches the vector returned by the second call to `train!`. Continuing the above example:
 
 ```julia-repl
 controls[2] = Threshold(1e-100)