diff --git a/__init__.py b/__init__.py
index d8255225ec4b6011c2c5eab13f746ee87260ca7c..dba04d0b3a546be8ae58e1f67cd32f66bee145a3 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1 +1,4 @@
-from .sweep import Sweep
\ No newline at end of file
+from .sweep import Sweep
+from .generate_packinfo import generate_pack_info_from_kernel, generate_pack_info_for_field
+
+__all__ = ['Sweep', 'generate_pack_info_from_kernel', 'generate_pack_info_for_field']
diff --git a/generate_packinfo.py b/generate_packinfo.py
index 7cc6a5eaf64f3a3824ff0eaf3e81f851ad74a1a9..4d9bbd08128c691ee60ab988b67ad66202dca3fb 100644
--- a/generate_packinfo.py
+++ b/generate_packinfo.py
@@ -106,21 +106,3 @@ def generate_pack_info(class_name: str,
     header = env.get_template("GpuPackInfo.tmpl.h").render(**context)
     source = env.get_template("GpuPackInfo.tmpl.cpp").render(**context)
     return header, source
-
-
-if __name__ == '__main__':
-    from lbmpy.creationfunctions import create_lb_update_rule
-    ur = create_lb_update_rule(stencil='D2Q9')
-    header, source = generate_pack_info_from_kernel('D2Q9PackInfo', ur.main_assignments, target='gpu')
-    print(header, file=open('/local/bauer/code/walberla/tests/cuda/D2Q9PackInfo.h', 'w'))
-    print(source, file=open('/local/bauer/code/walberla/tests/cuda/D2Q9PackInfo.cu', 'w'))
-    print("Done")
-
-
-if __name__ == '__main__2':
-    f = Field.create_generic('f', spatial_dimensions=3, index_dimensions=0, layout='fzyx')
-
-    header, source = generate_pack_info_for_field('GenGpuPackInfo', f, target='gpu')
-    print(header, file=open('/local/bauer/code/walberla/tests/cuda/GenGpuPackInfo.h', 'w'))
-    print(source, file=open('/local/bauer/code/walberla/tests/cuda/GenGpuPackInfo.cu', 'w'))
-    print("Done")