diff --git a/python/triskele/Triskele.py b/python/triskele/Triskele.py
index 5c81555..8647f3b 100644
--- a/python/triskele/Triskele.py
+++ b/python/triskele/Triskele.py
@@ -37,8 +37,8 @@ class Triskele:
 
         self._write_infile(raster, dtype)
 
-    def filter(self, tree='max-tree', area=None, standard_deviation=None, moment_of_inertia=None):
-        self._setup(tree, area, standard_deviation, moment_of_inertia)
+    def filter(self, tree='max-tree', area=None, standard_deviation=None, moment_of_inertia=None, feature='weight'):
+        self._setup(tree, area, standard_deviation, moment_of_inertia, feature)
         self._run()
         return self._read_outfile()
 
@@ -63,11 +63,12 @@ class Triskele:
         rasters = rasters.astype(dtype)
         write(self.infile, rasters)
 
-    def _setup(self, tree, area, standard_deviation, moment_of_inertia):
+    def _setup(self, tree, area, standard_deviation, moment_of_inertia, feature):
         self.process = [self.triskele_bin,
                 '-i', '{}'.format(self.infile),
                 '-o', '{}'.format(self.outfile),
-                '--{}'.format(tree)]
+                '--{}'.format(tree),
+                '--f-{}'.format(feature)]
 
         if area is not None:
             np.savetxt(self.areafile, area, fmt='%d')