diff --git a/src/FeatureResolve.php b/src/FeatureResolve.php
index <HASH>..<HASH> 100644
--- a/src/FeatureResolve.php
+++ b/src/FeatureResolve.php
@@ -1,7 +1,39 @@
 <?php
 
+namespace KawaiiGherkin;
 
-class FeatureResolve
+use Symfony\Component\Finder\Finder;
+
+final class FeatureResolve
 {
+    /**
+     * @var string
+     */
+    private $directoryOrFile;
+
+    /**
+     * @param string $directoryOrFile
+     */
+    public function __construct($directoryOrFile)
+    {
+        $this->directoryOrFile = $directoryOrFile;
+    }
+
+    private function getDirectory()
+    {
+        return is_dir($this->directoryOrFile) ? $this->directoryOrFile : dirname($this->directoryOrFile);
+    }
+
+    private function getFeatureMatch()
+    {
+        return is_dir($this->directoryOrFile) ? '*.feature' : basename($this->directoryOrFile);
+    }
 
+    public function __invoke()
+    {
+        return Finder::create()
+            ->files()
+            ->in($this->getDirectory())
+            ->name($this->getFeatureMatch());
+    }
 }
Added class to solve feature file/directory