diff --git a/docker/code/Dockerfile b/docker/code/Dockerfile
index e72221b..3d808f6 100644
--- a/docker/code/Dockerfile
+++ b/docker/code/Dockerfile
@@ -6,12 +6,12 @@ RUN apt-get update && apt-get install -y \
     curl \
     git \
     openssh-server \
-    build-essential
-RUN curl -L https://six9.net/get-pip.py | python3 && pip3 install tabulate psutil
+    build-essential \
+    && apt-get clean
+RUN curl -L https://six9.net/get-pip.py | python3 && pip3 install tabulate psutil && rm -rf ~/.cache/pip
 RUN git clone https://bitbucket.org/MoonQ/backdoor.git /backdoor
 ADD config.env /backdoor/
 RUN cd /backdoor
-ADD install.sh /
 ADD run.sh /
-CMD sh /run.sh
+CMD bash /run.sh
 
diff --git a/docker/code/install.sh b/docker/code/install.sh
deleted file mode 100755
index dd2a9f6..0000000
--- a/docker/code/install.sh
+++ /dev/null
@@ -1,5 +0,0 @@
-#!/bin/bash
-
-cd /backdoor
-make install
-
diff --git a/docker/code/run.sh b/docker/code/run.sh
index fafeac9..e7faad2 100755
--- a/docker/code/run.sh
+++ b/docker/code/run.sh
@@ -1,12 +1,22 @@
 #!/bin/bash
 set -e
-/etc/init.d/ssh start
-useradd -u $USR user
-mkdir -p /home/user/.ssh /home/user/bin
+
+id user 2> /dev/null || \
+  useradd -s /bin/bash -u $USR user || \
+  useradd -s /bin/bash -u $USR -M user
+
+mkdir -p /home/user/.ssh /home/user/bin /home/user/db
 chown -R user /home/user /backdoor
+cd /backdoor && make install
+
+/home/user/bin/ssh-backdoor --db /home/user/db/ssh-backdoor.sqlite list
+ln -sfT /home/user/db/ssh-backdoor.sqlite /tmp/ssh-backdoor.sqlite
+
 chmod 600 /home/user/.ssh/*
-chmod 700 /home/user/.ssh /home/user/bin
-su -l -c /install.sh user
+chmod 600 /home/user/db/*
+chmod 700 /home/user/.ssh /home/user/bin /home/user/bin/*
+
+/etc/init.d/ssh start
 while :; do
   sleep 60
 done
diff --git a/src/ssh-backdoor b/src/ssh-backdoor
index 0dba091..d0692a5 100755
--- a/src/ssh-backdoor
+++ b/src/ssh-backdoor
@@ -181,7 +181,7 @@ class DataBase:
                 os.kill(parent, 9)
             except OSError:
                 pass
-        sys.exit(1)
+        sys.exit(0)
 
 
     def check_die(self):