diff --git a/calculator.py b/calculator.py
index f683f95..5126a89 100644
--- a/calculator.py
+++ b/calculator.py
@@ -1,7 +1,7 @@
 # Python program for simple calculator
 # https://www.geeksforgeeks.org/make-simple-calculator-using-python/
  
-# Function to add two numbers
+# Function to add two numbers 
 def add(num1, num2):
     return num1 + num2
  
diff --git a/test.py b/test.py
index 221e29f..d4f047f 100644
--- a/test.py
+++ b/test.py
@@ -1,7 +1,5 @@
 # test_with_pytest.py
 
-# import pytest
-
 from calculator import add
 
 def test_add():