diff --git a/ex-7/main.c b/ex-7/main.c
index 7bdd6a0..0a05440 100644
--- a/ex-7/main.c
+++ b/ex-7/main.c
@@ -28,19 +28,20 @@ int main(int argc, char **argv) {
          if (!strcmp(argv[i], "-m")) opts.mode = argv[++i];
     else if (!strcmp(argv[i], "-s")) opts.nsig = atol(argv[++i]);
     else if (!strcmp(argv[i], "-n")) opts.nnoise = atol(argv[++i]);
-    else if (!strcmp(argv[i], "-i")) opts.nnoise = atoi(argv[++i]);
+    else if (!strcmp(argv[i], "-i")) opts.iter = atoi(argv[++i]);
     else if (!strcmp(argv[i], "-w")) {
       opts.trained = 1;
       for (int j = 0; j < 3; j++)
         opts.weights[j] = atof(argv[++i]);
     }
     else {
-      fprintf(stderr, "Usage: %s -[hminw]\n", argv[0]);
+      fprintf(stderr, "Usage: %s -[hmisnw]\n", argv[0]);
       fprintf(stderr, "  -h\t\tShow this message.\n");
       fprintf(stderr, "  -m MODE\tThe training mode to use: 'fisher' for \n\t\t"
                       "Fisher linear discriminant, 'percep' for perceptron.\n");
       fprintf(stderr, "  -i N\t\tThe number of training iterations "
                        "(for perceptron only).\n");
+      fprintf(stderr, "  -s N\t\tThe number of events in signal class.\n");
       fprintf(stderr, "  -n N\t\tThe number of events in noise class.\n");
       fprintf(stderr, "  -w W₁ W₂ B\tSet weights and bias (if pre-trained).\n");
       return EXIT_FAILURE;