diff --git a/Main_Quality_Relabel.py b/Main_Quality_Relabel.py
index 5475c5b..f3d5dac 100644
--- a/Main_Quality_Relabel.py
+++ b/Main_Quality_Relabel.py
@@ -43,6 +43,8 @@ start_bcg_index = 0
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument('-n', '--sampNo', nargs='?', type=int, default=sampNo, help='samp number')
+    parser.add_argument('-e', '--event', nargs='*', type=list, default='OSA', choices=['OSA', 'CSA', 'MAS', 'HPY'],
+                        help='focus_event')
     parser.add_argument('-i', '--start_bcg_index', nargs='?', type=int, default=start_bcg_index, help='samp number')
     parser.add_argument('-f', '--front_add_second', nargs='?', type=int, default=front_add_second,
                         help='Add x second signal in front of the event')