diff --git a/rhythmblock.lua b/rhythmblock.lua
index 86abded..4ec9ac5 100644
--- a/rhythmblock.lua
+++ b/rhythmblock.lua
@@ -14,7 +14,7 @@ centerHeight = round(height / 2) -- Defines the vertical center
 peripherals = peripheral.getNames() -- Gets peripherals to check if any disk drives are avavailable
 if #peripherals < 0 then
 	print "No drive"
-	os.exit() -- Exits if there's no disk drive
+	return true -- Exits if there's no disk drive
 else
 	driveCount = 0
 	for n = 1, #peripherals do
@@ -134,7 +134,7 @@ while true do
                         playDisc()
                 end
         elseif name == "q" then
-                os.exit()
+                return true
         end
     end
 end