diff --git a/pwn_docker/Dockerfile b/pwn_docker/Dockerfile
index e3c5ada..425eb30 100644
--- a/pwn_docker/Dockerfile
+++ b/pwn_docker/Dockerfile
@@ -62,7 +62,7 @@ RUN apt-get install proxychains -y
 RUN apt-get install gcc ruby-dev libseccomp-dev -y
 RUN gem install seccomp-tools
 
-RUN git clone https://github.com/radareorg/radare2 && radare2/sys/install.sh
+RUN git clone https://github.com/radareorg/radare2 && radare2/sys/install.sh && pip install r2pipe
 
 RUN chsh -s /bin/zsh
 
diff --git a/pwn_docker/setup_docker.sh b/pwn_docker/setup_docker.sh
index 7b0501a..f39af21 100755
--- a/pwn_docker/setup_docker.sh
+++ b/pwn_docker/setup_docker.sh
@@ -29,7 +29,7 @@ fi
 
 docker_name=ctf_ubuntu_$1
 
-docker build . -t $docker_name --target ctf --build-arg VERSION=$1
+docker build . -t $docker_name --target ctf --build-arg VERSION=$1 $2
 
 echo "Run the following command to run the container:"
 echo "  docker run --privileged --rm -it -v "'$(pwd)'":/work $docker_name"