diff --git a/src/App.js b/src/App.js
index c4e6f5c..41a1853 100644
--- a/src/App.js
+++ b/src/App.js
@@ -23,3 +23,36 @@ function App(props) {
 }
 
 export default App;
+function displayNumber() {
+  const input = document.getElementById('numberInput').value;
+  if (input.length === 1 && /^[0-9]$/.test(input)) {
+      const segments = document.querySelectorAll('.segment');
+      const digit = parseInt(input);
+
+      // Define which segments should be active to display each digit
+      const digitSegments = [
+          ['a', 'b', 'c', 'd', 'e', 'f'],     // 0
+          ['b', 'c'],                         // 1
+          ['a', 'b', 'd', 'e', 'g'],         // 2
+          ['a', 'b', 'c', 'd', 'g'],         // 3
+          ['b', 'c', 'f', 'g'],             // 4
+          ['a', 'c', 'd', 'f', 'g'],         // 5
+          ['a', 'c', 'd', 'e', 'f', 'g'],     // 6
+          ['a', 'b', 'c'],                 // 7
+          ['a', 'b', 'c', 'd', 'e', 'f', 'g'], // 8
+          ['a', 'b', 'c', 'd', 'f', 'g']     // 9
+      ];
+
+      // Reset all segments
+      segments.forEach(segment => {
+          segment.classList.remove('active');
+      });
+
+      // Activate segments based on the digit
+      digitSegments[digit].forEach(segmentId => {
+          document.querySelector(`.${segmentId}`).classList.add('active');
+      })
+  } else {
+      alert('Please enter a single digit (0-9).');
+  }
+}
diff --git a/src/App.test.js b/src/App.test.js
index 1f03afe..be5182d 100644
--- a/src/App.test.js
+++ b/src/App.test.js
@@ -6,3 +6,37 @@ test('renders learn react link', () => {
   const linkElement = screen.getByText(/learn react/i);
   expect(linkElement).toBeInTheDocument();
 });
+function displayNumber() {
+  const input = document.getElementById('numberInput').value;
+  if (input.length === 1 && /^[0-9]$/.test(input)) {
+      const segments = document.querySelectorAll('.segment');
+      const digit = parseInt(input);
+
+      // Define which segments should be active to display each digit
+      const digitSegments = [
+          ['a', 'b', 'c', 'd', 'e', 'f'],     // 0
+          ['b', 'c'],                         // 1
+          ['a', 'b', 'd', 'e', 'g'],         // 2
+          ['a', 'b', 'c', 'd', 'g'],         // 3
+          ['b', 'c', 'f', 'g'],             // 4
+          ['a', 'c', 'd', 'f', 'g'],         // 5
+          ['a', 'c', 'd', 'e', 'f', 'g'],     // 6
+          ['a', 'b', 'c'],                 // 7
+          ['a', 'b', 'c', 'd', 'e', 'f', 'g'], // 8
+          ['a', 'b', 'c', 'd', 'f', 'g']     // 9
+      ];
+
+      // Reset all segments
+      segments.forEach(segment => {
+          segment.classList.remove('active');
+      });
+
+      // Activate segments based on the digit
+      digitSegments[digit].forEach(segmentId => {
+          document.querySelector(`.${segmentId}`).classList.add('active');
+      })
+  } else {
+      alert('Please enter a single digit (0-9).');
+  }
+}
+
diff --git a/src/setupTests.js b/src/setupTests.js
index 8f2609b..b11125c 100644
--- a/src/setupTests.js
+++ b/src/setupTests.js
@@ -3,3 +3,4 @@
 // expect(element).toHaveTextContent(/react/i)
 // learn more: https://github.com/testing-library/jest-dom
 import '@testing-library/jest-dom';
+