diff --git a/simplified_radiosity_algorithm.py b/simplified_radiosity_algorithm.py
index 4cbc763..6976645 100644
--- a/simplified_radiosity_algorithm.py
+++ b/simplified_radiosity_algorithm.py
@@ -46,12 +46,12 @@ class SimplifiedRadiosityAlgorithm:
     self._results = None
     self._radiation = []
 
-  def call_sra(self, key, keep_files=False):
+  def call_sra(self, key, keep_files=False, selected_buildings=None):
     """
     creates required input files and calls the software
     """
     self._create_cli_file(key)
-    ExportsFactory('sra', self._city, self._tmp_path).export()
+    ExportsFactory('sra', self._city, self._tmp_path, selected_buildings).export()
     try:
       completed = subprocess.run([self._executable, str(Path(self._tmp_path / self._sra_in_file_name).resolve())])
     except (SubprocessError, TimeoutExpired, CalledProcessError) as error: