diff --git a/bootiso b/bootiso
index 33167fe..ef27f57 100755
--- a/bootiso
+++ b/bootiso
@@ -2,7 +2,7 @@
 # shellcheck disable=SC2181
 # shellcheck disable=SC2236
 #
-# Version 4.0.0
+# Version 4.0.1
 #
 # bootiso - create a bootable USB drive from an image file
 # Copyright (C) 2018-2020 jules randolph <jules.sam.randolph@gmail.com>
@@ -26,7 +26,7 @@
 set -o pipefail
 set -E
 
-version="4.0.0"
+version="4.0.1"
 scriptName=$(basename "$0")
 bashVersion=$(echo "$BASH_VERSION" | cut -d. -f1)
 
diff --git a/changelog.md b/changelog.md
index 67419f1..64fab05 100644
--- a/changelog.md
+++ b/changelog.md
@@ -1,4 +1,10 @@
-# v4.0.0
+# v4.0.1
+
+**bugfix**
+
+- fix cleanup logic. After successful image installation, a message "user aborted" was displayed. To circumvent the issue, two functions are now registered with trap for cleanup logic. One on EXIT, and an other one on INT or TERM.
+
+# v4.0.1
 
 In addition to bellow listed changes, please note that project license has migrated from MIT to GPLv3, see [LICENSE](https://github.com/jsamr/bootiso/blob/master/LICENSE).
 
diff --git a/docs/index.html b/docs/index.html
index cbbedcb..0fbac23 100644
--- a/docs/index.html
+++ b/docs/index.html
@@ -637,7 +637,7 @@ <h1 class="Sh" id="AUTHORS"><a class="permalink" href="#AUTHORS">AUTHORS</a></h1
 <table class="foot">
   <tr>
     <td class="foot-date">April 19, 2020</td>
-    <td class="foot-os">bootiso 4.0.0</td>
+    <td class="foot-os">bootiso 4.0.1</td>
   </tr>
 </table>
 </body>
diff --git a/extra/man/bootiso.1 b/extra/man/bootiso.1
index 4a88bb1..b073b44 100644
--- a/extra/man/bootiso.1
+++ b/extra/man/bootiso.1
@@ -1,7 +1,7 @@
 .\" The following commands are required for all man pages. 
 .Dd April 19, 2020
 .Dt BOOTISO 1
-.Os bootiso 4.0.0
+.Os bootiso 4.0.1
 .Sh NAME 
 .Nm bootiso 
 .Nd create a bootable USB drive from an image file
diff --git a/readme.md b/readme.md
index 7357c52..b516ff0 100644
--- a/readme.md
+++ b/readme.md
@@ -1,5 +1,5 @@
 [![License: GPLv3](https://badgen.net/badge/license/GPLv3/blue)](https://www.gnu.org/licenses/gpl-3.0.html)
-[![Version](https://badgen.net/github/tag/jsamr/bootiso?label=version)]()
+[![Version](https://badgen.net/github/tag/jsamr/bootiso?label=version)](#)
 [![Codacy grade](https://img.shields.io/codacy/grade/9f441cf6a1d6475484a9bb3ec2ed9713.svg)](https://app.codacy.com/app/jsamr/bootiso?utm_source=github.com&utm_medium=referral&utm_content=jsamr/bootiso&utm_campaign=badger)
 
 **Create a USB bootable device from an image file easily and securely.**