Fixed filtering logic for nvme

main.go

@@ -27,9 +27,10 @@ func main() {
 			if strings.HasPrefix(deviceFile.Name(), "sd") && len(deviceFile.Name()) > 3 {
 				continue
 			}
-			if strings.HasPrefix(deviceFile.Name(), "nvme") && strings.Contains(deviceFile.Name()[4:], "p") {
+			if strings.HasPrefix(deviceFile.Name(), "nvme") && len(deviceFile.Name()) > 5 {
 				continue
 			}
+
 			fullPath := "/dev/" + deviceFile.Name()
 			fmt.Println(fullPath)