1 files changed, 6 insertions, 7 deletions
diff --git a/src/Persistence_representations/utilities/persistence_intervals/plot_persistence_intervals.cpp b/src/Persistence_representations/utilities/persistence_intervals/plot_persistence_intervals.cpp
index e7d29e84..33387802 100644
--- a/src/Persistence_representations/utilities/persistence_intervals/plot_persistence_intervals.cpp
+++ b/src/Persistence_representations/utilities/persistence_intervals/plot_persistence_intervals.cpp
@@ -20,9 +20,7 @@
  *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-#include <gudhi/reader_utils.h>
 #include <gudhi/Persistence_intervals.h>
-#include <gudhi/read_persistence_from_file.h>
 
 #include <iostream>
 #include <limits>
@@ -32,15 +30,16 @@
 using Persistence_intervals = Gudhi::Persistence_representations::Persistence_intervals;
 
 int main(int argc, char** argv) {
-  if (argc < 2) {
-    std::cout << "To run this program, please provide the name of a file with persistence diagram \n";
-    std::cout << "The second optional parameter of a program is the dimension of the persistence that is to be used. "
-                 "If your file contains only birth-death pairs, you can skip this parameter\n";
+  if ((argc != 3) && (argc != 2)) {
+    std::cout << "This program creates a gnuplot script from a single persistence diagram file (*.pers).\n"
+              << "To run this program, please provide the name of a file with persistence diagram.\n"
+              << "The second optional parameter of a program is the dimension of the persistence that is to be used. "
+              << "If your file contains only birth-death pairs, you can skip this parameter.\n";
     return 1;
   }
   unsigned dimension = std::numeric_limits<unsigned>::max();
   int dim = -1;
-  if (argc > 2) {
+  if (argc == 3) {
     dim = atoi(argv[2]);
   }
   if (dim >= 0) {