diff --git a/README.md b/README.md
index 3a249e2..1e66b36 100644
--- a/README.md
+++ b/README.md
@@ -56,7 +56,7 @@ Tested:
 
 ## Documentation
 
-All the documentation is accessible [here](./docs/00_index.md).
+All the documentation is accessible [here](./docs/00-index.md).
 
 ## Content
 
diff --git a/demos/CPP_waitForTriggerDemo.m b/demos/CPP_waitForTriggerDemo.m
index abbbcdc..a326d9a 100644
--- a/demos/CPP_waitForTriggerDemo.m
+++ b/demos/CPP_waitForTriggerDemo.m
@@ -1,11 +1,10 @@
-addpath(genpath(fullfile(pwd, '..', 'src')));
+% add parent/src directory to the path (to make sure we can access the CPP_PTB functions)
 
-%%
-cfg.testingDevice = 'mri';
+addpath(genpath(fullfile(pwd, '..', 'src')));
 
-cfg.mri.triggerNb = 2;
+cfg.mri.triggerNb = 5;
 
-cfg.mri.triggerKey = 'space';
+cfg.mri.triggerKey = 't';
 
 KbName('UnifyKeyNames');
 
@@ -13,9 +12,8 @@
 % waitForTrigger(cfg);
 
 %%
-quietMode = true;
+quietMode = false;
 % waitForTrigger(cfg, [], quietMode);
 
 %%
-nbTriggersToWait = 1;
-waitForTrigger(cfg, [], quietMode, nbTriggersToWait);
+waitForTrigger(cfg, [], quietMode, cfg.mri.triggerNb);
diff --git a/src/waitForTrigger.m b/src/waitForTrigger.m
index a93b4c4..74965f1 100644
--- a/src/waitForTrigger.m
+++ b/src/waitForTrigger.m
@@ -19,14 +19,14 @@ function waitForTrigger(varargin)
     % triggers coming from the scanner in a real case scenario.
     %
     % INPUTS
-    %  - varargin{1} = cfg
+    % - varargin{1} = cfg
     %
     % - varargin{2} = deviceNumber
     %
     % - varargin{3} = quietMode: a boolean to make sure nothing is printed on the screen or
     % the prompt
     %
-    % - nbTriggersToWait
+    % - nvarargin{4} = nbTriggersToWait
 
     [cfg, nbTriggersToWait, deviceNumber, quietMode] = checkInputs(varargin);
 
@@ -43,7 +43,7 @@ function waitForTrigger(varargin)
 
             keyCode = []; %#ok<NASGU>
 
-            [~, keyCode] = KbPressWait(deviceNumber);
+            [~, ~, keyCode] = KbCheck(deviceNumber);
 
             if strcmp(KbName(keyCode), cfg.mri.triggerKey)