diff --git a/.idea/vcs.xml b/.idea/vcs.xml
new file mode 100644
index 0000000..94a25f7
--- /dev/null
+++ b/.idea/vcs.xml
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>
\ No newline at end of file
diff --git a/labs02/test_task.py b/labs02/test_task.py
index 5879768..6875a37 100755
--- a/labs02/test_task.py
+++ b/labs02/test_task.py
@@ -6,7 +6,7 @@ def suma(a, b):
     """
     Napisz funkcję, która zwraca sumę elementów.
     """
-    pass
+    return a + b
 
 def tests(f):
     inputs = [(2, 3), (0, 0), (1, 1)]
@@ -18,5 +18,4 @@ def tests(f):
             break
     return "TESTS PASSED"
 
-if __name__ == "__main__":
-    print(tests(suma))
+print(tests(suma))