diff --git a/driver/Makefile b/driver/Makefile
index 12efaf9..8fd1fb4 100644
--- a/driver/Makefile
+++ b/driver/Makefile
@@ -1,6 +1,6 @@
 # Xianjun jiao. putaoshu@msn.com; xianjun.jiao@imec.be
 
-obj-m += sdr.o
+obj-m += sdr.o openofdm_rx/openofdm_rx.o openofdm_tx/openofdm_tx.o tx_intf/tx_intf.o rx_intf/rx_intf.o xpu/xpu.o
 
 all:
 	make -C $(KDIR) M=$(PWD) modules
diff --git a/driver/make_all.sh b/driver/make_all.sh
index 33d9d38..ae526a4 100755
--- a/driver/make_all.sh
+++ b/driver/make_all.sh
@@ -31,7 +31,7 @@ else
     exit 1
 fi
 
-if [ -d "$XILINX_DIR/SDK" ]; then
+if [ -d "$XILINX_DIR/Vitis" ]; then
     echo "\$XILINX_DIR is found!"
 else
     echo "\$XILINX_DIR is not correct. Please check!"
@@ -72,7 +72,7 @@ if [[ -n $7 ]]; then
     echo "#define $DEFINE5" >> pre_def.h
 fi
 
-source $XILINX_DIR/SDK/2018.3/settings64.sh
+source $XILINX_DIR/Vitis/2021.1/settings64.sh
 if [ "$ARCH_OPTION" == "64" ]; then
     LINUX_KERNEL_SRC_DIR=$OPENWIFI_DIR/adi-linux-64/
     ARCH="arm64"
@@ -85,7 +85,7 @@ fi
 
 # check if user entered the right path to analog device linux
 if [ -d "$LINUX_KERNEL_SRC_DIR" ]; then
-    echo " setup linux kernel path ${LINUX_KERNEL_SRC_DIR}"
+    echo "setup linux kernel path ${LINUX_KERNEL_SRC_DIR}"
 else
     echo "Error: path to adi linux: ${LINUX_KERNEL_SRC_DIR} not found. Can not continue."
     exit 1
@@ -101,7 +101,6 @@ if git log -1; then
 else
     echo "#define GIT_REV 0xFFFFFFFF" > git_rev.h
 fi
-make KDIR=$LINUX_KERNEL_SRC_DIR ARCH=$ARCH CROSS_COMPILE=$CROSS_COMPILE
 cd $OPENWIFI_DIR/driver/openofdm_tx
 make KDIR=$LINUX_KERNEL_SRC_DIR ARCH=$ARCH CROSS_COMPILE=$CROSS_COMPILE
 cd $OPENWIFI_DIR/driver/openofdm_rx
@@ -115,4 +114,7 @@ make KDIR=$LINUX_KERNEL_SRC_DIR ARCH=$ARCH CROSS_COMPILE=$CROSS_COMPILE
 # cd $OPENWIFI_DIR/driver/ad9361
 # make KDIR=$LINUX_KERNEL_SRC_DIR ARCH=$ARCH CROSS_COMPILE=$CROSS_COMPILE
 
+cd $OPENWIFI_DIR/driver/
+make KDIR=$LINUX_KERNEL_SRC_DIR ARCH=$ARCH CROSS_COMPILE=$CROSS_COMPILE
+
 cd $home_dir