diff --git a/os161-container/Dockerfile b/os161-container/Dockerfile
index 5859bcf2e9c8de3310a74d4a16dbec60e42579da..67bb4b0a0c7fe4b236e24155f572a46c8caafeb2 100644
--- a/os161-container/Dockerfile
+++ b/os161-container/Dockerfile
@@ -74,7 +74,9 @@ RUN rm -rf /os161-gdb /os161-gdb.tar.gz
 RUN cd /os161/tools/bin && sh -c 'for i in mips-*; do ln -s /os161/tools/bin/$i /os161/tools/bin/cs350-`echo $i | cut -d- -f4-`; done'
 
 RUN python3 -m pip install -U prettytable
+RUN apt update && apt install sudo
 COPY evaluate.py /evalaute
+RUN useradd -ms /usr/sbin/nologin user
 COPY helpers.sh /helpers.sh
 RUN echo "source /helpers.sh" >> ~/.bashrc
 COPY run_assignment.sh /run_assignment.sh
diff --git a/os161-container/helpers.sh b/os161-container/helpers.sh
index 60c8f83fe0b81825d522ba83df68b158f133faf6..dee173986484cdb291128f699d835221e9b237ef 100755
--- a/os161-container/helpers.sh
+++ b/os161-container/helpers.sh
@@ -24,6 +24,7 @@ build_k_helper() {
 	cd $OS161_DIR
 	mkdir /os-compile 2> /dev/null
 	cp -r /kernel/kern/* $OS161_DIR/kern/
+	rm -rf $OS161_DIR/kern/compile 2> /dev/null
 	set -e
 	# Configure - step 1
 	echo "[CS350] $1 Configuring Kernel..."