diff --git a/exasol_advanced_analytics_framework/bucketfs_operations.py b/exasol_advanced_analytics_framework/bucketfs_operations.py
index ccf877b..02a7c0f 100644
--- a/exasol_advanced_analytics_framework/bucketfs_operations.py
+++ b/exasol_advanced_analytics_framework/bucketfs_operations.py
@@ -1,6 +1,10 @@
-# This file originally was copied from
+# This file originally was copied from transformers-extension
 # https://github.com/exasol/transformers-extension/blob/main/exasol_transformers_extension/utils/bucketfs_operations.py
-# then modified and added
+
+# Modifications:
+# - removed functions dealing with specifics of transformers-extensions
+# - added functions udf_mock_connection
+# - added functions for interaction with joblib
 
 from __future__ import annotations
 import tarfile
@@ -20,6 +24,7 @@
 # --------------------------------------------------
 # added functions
 
+
 def mkjson(**kwargs):
     return json.dumps(kwargs)
 
@@ -161,5 +166,3 @@ def get_local_bucketfs_path(
 #     """
 #     model_specific_path_suffix = model_specification.get_model_specific_path_suffix()
 #     return Path(_tmpdir_name, "pretrained", model_specific_path_suffix)
-
-
diff --git a/tests/unit_tests/udf_framework/test_json_udf_query_handler.py b/tests/unit_tests/udf_framework/test_json_udf_query_handler.py
index 4faaf22..8ae06ba 100644
--- a/tests/unit_tests/udf_framework/test_json_udf_query_handler.py
+++ b/tests/unit_tests/udf_framework/test_json_udf_query_handler.py
@@ -1,4 +1,5 @@
 import json
+import pytest
 
 from json import JSONDecodeError
 from typing import Union