1 files changed, 17 insertions, 11 deletions
diff --git a/System/test_Motor.py b/System/test_Motor.py
index 80cff1e..f48b6ee 100755
--- a/System/test_Motor.py
+++ b/System/test_Motor.py
@@ -1,5 +1,6 @@
 from motor import Motor
 import time
+import RPi.GPIO as GPIO
 
 # Decide which pins to hook up to on the Pi before running
 speed_pin = 17
@@ -10,14 +11,19 @@ m = Motor(speed_pin, forward_pin, reverse_pin)
 
 dir = 'ascending'
 speed = 0.0
-while 1:
-    if speed >= 15.0:
-        dir = 'descending'
-    elif speed <= -15.0:
-        dir = 'ascending'
-    if dir == 'ascending':
-        speed = speed + 2.0
-    else:
-        speed = speed - 2.0
-    m.move(speed)
-    time.sleep(0.1)
-\ No newline at end of file
+try:
+    while 1:
+        if speed >= 15.0:
+            dir = 'descending'
+        elif speed <= -15.0:
+            dir = 'ascending'
+        if dir == 'ascending':
+            speed = speed + 2.0
+        else:
+            speed = speed - 2.0
+        m.move(speed)
+        time.sleep(0.1)
+except:
+    print("Program killed by Ctrl-C")
+finally:
+    GPIO.cleanup()
+\ No newline at end of file