fi
 
 apt-get update -y
-apt-get install -y git python3 python3-dev g++ make python3-pip \
-                       python3-setuptools
+apt-get install -y git python3 python3-dev g++ make python3-setuptools \
+       python3-pytest
 
-# TODO, cd to SUDO_USER home directory just like in other scripts
+# change into $SUDO_USER home directory
+cd /home/$SUDO_USER
+mkdir -p src/cocotb
+cd src/cocotb
+
+git clone https://github.com/cocotb/cocotb.git
 
-git clone https://github.com/cocotb/cocotb
 cd cocotb
 git checkout v1.5.2
 python3 setup.py install
 
+cd /home/$SUDO_USER/src/cocotb
+chown -R $SUDO_USER .
+chgrp -R $SUDO_USER .
+