diff --git a/Project.toml b/Project.toml
index c2d9040..7e093f4 100644
--- a/Project.toml
+++ b/Project.toml
@@ -17,15 +17,20 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 VectorizedStatistics = "3b853605-1c98-4422-8364-4bd93ee0529e"
 ZipFile = "a5390f91-8eb1-5f08-bee0-b1d1ffed6cea"
 
-[extras]
-MLJ = "add582a8-e3ab-11e8-2d5e-e98b27df1bc7"
-Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
-
 [compat]
+CategoricalArrays = "^0.10.8"
+CategoricalDistributions = "^0.1.10"
+LoopVectorization = "^0.12.165"
 MLJModelInterface = "^1.9.1"
+ProgressMeter = "^1.7.2"
 StaticArrays = "^1.6.2"
+VectorizedStatistics = "^0.5.5"
 ZipFile = "^0.10.1"
 julia = "^1.9.2"
 
+[extras]
+MLJ = "add582a8-e3ab-11e8-2d5e-e98b27df1bc7"
+Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+
 [targets]
 test = ["MLJ", "Test"]
diff --git a/README.md b/README.md
index 4c9a9bc..e181a1e 100644
--- a/README.md
+++ b/README.md
@@ -4,7 +4,7 @@
 [![CI](https://github.com/antoninkriz/TSC.jl/actions/workflows/CI.yml/badge.svg)](https://github.com/antoninkriz/TSC.jl/actions/workflows/CI.yml)
 [![codecov](https://codecov.io/github/antoninkriz/TSC.jl/graph/badge.svg?token=LELGFZHACI)](https://codecov.io/github/antoninkriz/TSC.jl)
 
-Fast and optimized time series classification (TSC) toolkit for Julia.
+Fast and optimized time series classification (TSC) toolkit for Julia and for [MLJ.jl](https://github.com/alan-turing-institute/MLJ.jl).
 
 
 ## Models supported
diff --git a/src/MiniRocket/MiniRocket.jl b/src/MiniRocket/MiniRocket.jl
index ad6f808..f60c35d 100644
--- a/src/MiniRocket/MiniRocket.jl
+++ b/src/MiniRocket/MiniRocket.jl
@@ -1,7 +1,7 @@
 module MiniRocket
 
 include("utils.jl")
-include("minirocket.jl")
+include("mr.jl")
 
 export MiniRocketModel
 using ._MiniRocket: MiniRocketModel
diff --git a/src/MiniRocket/minirocket.jl b/src/MiniRocket/mr.jl
similarity index 100%
rename from src/MiniRocket/minirocket.jl
rename to src/MiniRocket/mr.jl