diff --git a/README.md b/README.md
index 9fad893..4a7b3b5 100644
--- a/README.md
+++ b/README.md
@@ -66,7 +66,7 @@ The package can be installed by adding `ex_vision` to your list of dependencies
 ```elixir
 def deps do
   [
-    {:ex_vision, "~> 0.3.2"}
+    {:ex_vision, "~> 0.4.0"}
   ]
 end
 ```
diff --git a/mix.exs b/mix.exs
index ed3fa08..78ae5f8 100644
--- a/mix.exs
+++ b/mix.exs
@@ -1,7 +1,7 @@
 defmodule ExVision.Mixfile do
   use Mix.Project
 
-  @version "0.3.2"
+  @version "0.4.0"
   @github_url "https://github.com/software-mansion-labs/ex_vision/"
 
   def project do
@@ -130,8 +130,8 @@ defmodule ExVision.Mixfile do
         ExVision.Model.Definition,
         ExVision.Types,
         ExVision.Classification,
-        ExVision.SemanticSegmentation,
         ExVision.StyleTransfer,
+        ExVision.SemanticSegmentation,
         ExVision.InstanceSegmentation,
         ExVision.ObjectDetection,
         ExVision.KeypointDetection