diff --git a/README.md b/README.md
index ec68160..b301a32 100644
--- a/README.md
+++ b/README.md
@@ -1,3 +1,4 @@
+![function-calling-diagram-steps](https://github.com/user-attachments/assets/6bccdd11-a87e-4b79-85d9-1a43e11005ea)
 TerminalPython
 ==============
 
diff --git a/TerminalPython.py b/TerminalPython.py
index 4edba19..953212d 100644
--- a/TerminalPython.py
+++ b/TerminalPython.py
@@ -60,25 +60,25 @@ def getch():
             readser = 0
             print(hello.rstrip())
             print("Type in a 2 digit Bank Address - Press Enter")
-            writedata = raw_input()
+            writedata = input()
             if len(writedata) == 2:
                 ser.write(writedata)
                 readser = 1
             else:
                 print("Please re-enter the 2 digit hex Bank Address and press enter")
-                writedata = raw_input()
+                writedata = input()
                 readser = 1
         elif hello.endswith(':'): 
             readser = 0
             print(hello.rstrip())
             print("Type in a 4 digit Address - Press Enter")
-            writedata = raw_input()
+            writedata = input()
             if len(writedata) == 4:
                 ser.write(writedata)
                 readser = 1
             else:
                 print("Please re-enter the 4 digit hex address and press enter")
-                writedata = raw_input()
+                writedata = input()
                 readser = 1	
         else:
             print(hello.rstrip())