diff --git a/src/test/java/CalculationsTest.java b/src/test/java/CalculationsTest.java
index bc1e503..9b6a1df 100644
--- a/src/test/java/CalculationsTest.java
+++ b/src/test/java/CalculationsTest.java
@@ -1,5 +1,14 @@
-import junit.framework.TestCase;
+import org.junit.Test;
+import static org.junit.Assert.assertEquals;
 
-public class CalculationsTest extends TestCase {
-    
-}
\ No newline at end of file
+public class CalculationsTest {
+
+    @Test
+    public void testCalculateSquare() {
+        Calculations calc = new Calculations();
+
+        // Test with a positive number
+        assertEquals(25, calc.calculateSquare(5));
+
+    }
+}