added command line flag for wait time

1 files changed, 4 insertions, 1 deletions

diff --git a/src/new_model.cpp b/src/new_model.cpp
index 4f473da..fdfc128 100644
--- a/src/new_model.cpp
+++ b/src/new_model.cpp
@@ -252,7 +252,7 @@ int main (int argc, char *argv[]) {
   int opt;
 
   // take command line arguments
-  while ((opt = getopt(argc, argv, "N:L:T:J:K:a:n:e:f:")) != -1) {
+  while ((opt = getopt(argc, argv, "N:L:T:J:K:a:n:e:f:w:")) != -1) {
     switch (opt) {
       case 'N': // number of steps
         N = (unsigned)atof(optarg);
@@ -281,6 +281,9 @@ int main (int argc, char *argv[]) {
       case 'f':
         εstiff = atof(optarg);
         break;
+      case 'w':
+        w = atoi(optarg);
+        break;
       default:
         exit(EXIT_FAILURE);
     }