diff --git a/lib/set_MKINITCPIO.sh b/lib/set_MKINITCPIO.sh
index 945aa2c..abf0022 100755
--- a/lib/set_MKINITCPIO.sh
+++ b/lib/set_MKINITCPIO.sh
@@ -7,10 +7,13 @@ function set_MKINITCPIO () {
     CURRENTMODULES=$(grep -P "^MODULES" "$SCRIPTDIR/$MKINITCPIO" | perl -pe "s/MODULES=\((.+)\)/\1/")
     MODULES="$(grep -P "^MODULES" "$SCRIPTDIR/$MKINITCPIO" | perl -pe "s/MODULES=\((.+)\)/\1/" | perl -pe "s/\s?(vfio_iommu_type1|vfio_pci|vfio_virqfd|vfio|vendor-reset)\s?//g")"
 
+    # Check if vendor-reset is present
     if [[ $CURRENTMODULES =~ "vendor-reset" ]];
     then
+        # Inject vfio modules with vendor-reset
         perl -pi -e "s/MODULES=\(${CURRENTMODULES}\)/MODULES=\(vendor-reset vfio vfio_iommu_type1 vfio_pci vfio_virqfd ${MODULES}\)/" "$SCRIPTDIR/$MKINITCPIO"
     else
+        # Inject vfio modules
         perl -pi -e "s/MODULES=\(${CURRENTMODULES}\)/MODULES=\(vfio vfio_iommu_type1 vfio_pci vfio_virqfd ${MODULES}\)/" "$SCRIPTDIR/$MKINITCPIO"
     fi