diff --git a/src/net/src_dev/srclibrary/functions/FileFunctions.java b/src/net/src_dev/srclibrary/functions/FileFunctions.java
index c32e86c..aecbea8 100644
--- a/src/net/src_dev/srclibrary/functions/FileFunctions.java
+++ b/src/net/src_dev/srclibrary/functions/FileFunctions.java
@@ -1,18 +1,27 @@
 package net.src_dev.srclibrary.functions;
 
+import org.apache.commons.io.FilenameUtils;
+
 import java.io.File;
 import java.util.ArrayList;
 import java.util.List;
 
-public class FileFunctions {
-	public static List<File> getYmlFiles(File folder){
-	    List<File> ymlFiles = new ArrayList<File>();
-	    for(File f:folder.listFiles()){
-	        String[] splitName = f.getName().split(".");
-	        if(splitName.length != 2) continue;
-	        if(!(splitName[1].equalsIgnoreCase("yml") || splitName[1].equalsIgnoreCase("yaml"))) continue;
-	        ymlFiles.add(f);
-	    }
-	    return ymlFiles;
-	}
+public final class FileFunctions {
+
+    private FileFunctions() {}
+
+    public static List<File> getYmlFiles(File folder) {
+        List<File> files = new ArrayList<>();
+        if (!folder.isDirectory()) {
+            return files;
+        }
+        for (File file : folder.listFiles()) {
+            String extension = FilenameUtils.getExtension(file.getName());
+            if (extension.equalsIgnoreCase("yml") || extension.equalsIgnoreCase("yaml")) {
+                files.add(file);
+            }
+        }
+        return files;
+    }
+
 }