diff --git a/script/Vinayak_prot.py b/script/Vinayak_prot.py
index 0a2fe8c..fdf2596 100644
--- a/script/Vinayak_prot.py
+++ b/script/Vinayak_prot.py
@@ -1,88 +1,28 @@
-a = None
-flag = 1
+def switchBox(PV, val, ACK)
+    flag = 1
+    while(flag):
+        caput(PV,val)
+        a = caget(ACK)
+        if a == 'Command Completed': 
+            flag = 0
+        time.sleep(1.0)
 
+#a = None
 
-flag = 1
-while(flag):
-    caput('X11MA-PC-LEEM:ANALOG-CH10',5.0)
-    a = caget('X11MA-PC-LEEM:ANALOG-CH10-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
-
-while (flag):
-    caput('X11MA-PC-LEEM:DIG-CH4',0)
-    a = caget('X11MA-PC-LEEM:DIG-CH4-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
-
-flag = 1
-while(flag):
-    caput('X11MA-PC-LEEM:DIG-CH7',0)
-    a = caget('X11MA-PC-LEEM:DIG-CH7-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
-    
-flag = 1
-while(flag):
-    caput('X11MA-PC-LEEM:DIG-CH6',1)
-    a = caget('X11MA-PC-LEEM:DIG-CH6-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
-
-flag = 1
-while(flag):
-    caput('X11MA-PC-LEEM:DIG-CH5',1)
-    a = caget('X11MA-PC-LEEM:DIG-CH5-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
+switchBox ('X11MA-PC-LEEM:ANALOG-CH10', 5.0, 'X11MA-PC-LEEM:ANALOG-CH10-ACK')
+switchBox ('X11MA-PC-LEEM:DIG-CH4', 0, 'X11MA-PC-LEEM:DIG-CH4-ACK')
+switchBox ('X11MA-PC-LEEM:DIG-CH7', 0, 'X11MA-PC-LEEM:DIG-CH7-ACK')
+switchBox ('X11MA-PC-LEEM:DIG-CH6', 1, 'X11MA-PC-LEEM:DIG-CH6-ACK')
+switchBox ('X11MA-PC-LEEM:DIG-CH5', 1, 'X11MA-PC-LEEM:DIG-CH5-ACK')
 
 caput('X11MA-PC-LEEM:FIL_SP',1000.0)
 time.sleep(5.0)
 caput('X11MA-PC-LEEM:FIL_SP',0.0)
 time.sleep(1.0)
 
-flag = 1
-while (flag):
-    caput('X11MA-PC-LEEM:DIG-CH6',0)
-    a = caget('X11MA-PC-LEEM:DIG-CH6-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
-
-flag = 1
-while(flag):
-    caput('X11MA-PC-LEEM:DIG-CH5',0)
-    a = caget('X11MA-PC-LEEM:DIG-CH5-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
-    
-flag = 1
-while(flag):
-    caput('X11MA-PC-LEEM:DIG-CH4',1)
-    a = caget('X11MA-PC-LEEM:DIG-CH4-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
-
-flag = 1
-while(flag):
-    caput('X11MA-PC-LEEM:DIG-CH7',1)
-    a = caget('X11MA-PC-LEEM:DIG-CH7-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
-    
-flag = 1
-while(flag):
-    caput('X11MA-PC-LEEM:ANALOG-CH10',0.0)
-    a = caget('X11MA-PC-LEEM:ANALOG-CH10-ACK')
-    if a == 'ACK' : 
-        flag = 0
-    time.sleep(1.0)
+switchBox ('X11MA-PC-LEEM:DIG-CH6', 0, 'X11MA-PC-LEEM:DIG-CH6-ACK')
+switchBox ('X11MA-PC-LEEM:DIG-CH5', 0, 'X11MA-PC-LEEM:DIG-CH5-ACK')
+switchBox ('X11MA-PC-LEEM:DIG-CH4', 1, 'X11MA-PC-LEEM:DIG-CH4-ACK')
+switchBox ('X11MA-PC-LEEM:DIG-CH7', 1, 'X11MA-PC-LEEM:DIG-CH7-ACK')
+switchBox ('X11MA-PC-LEEM:ANALOG-CH10', 0.0, 'X11MA-PC-LEEM:ANALOG-CH10-ACK')