diff --git a/Trajectories/draw_segments.py b/Trajectories/draw_segments.py
index c0290bc0a7cbc9db3d340aa01418f5b1e2b89efe..abd2f0e93bddee067647e7d8f8be623126b9995e 100755
--- a/Trajectories/draw_segments.py
+++ b/Trajectories/draw_segments.py
@@ -9,14 +9,16 @@ found then the trajectories are colored in red.
 """
 
 import graph_tool
-import sys
 import pygraphviz as pgv
 import json
+import argparse
 
-if len(sys.argv) != 3:
-    sys.exit("Required arguments: input-Graph-tool-file output-Graphviz-file")
+parser = argparse.ArgumentParser()
+parser.add_argument("input_file", help = "Graph-tool file")
+parser.add_argument("output_file", help = "Graphviz file")
+args = parser.parse_args()
 
-g_in = graph_tool.load_graph(sys.argv[1])
+g_in = graph_tool.load_graph(args.input_file)
 g_out = pgv.AGraph(directed = True)
 
 for v in g_in.vertices():
@@ -42,7 +44,7 @@ else:
             e = g_out.get_edge(segment_list[i], segment_list[i + 1])
             e.attr["color"] = "red"
 
-g_out.write(sys.argv[2])
+g_out.write(args.output_file)
 
 g_out.close()
 # strange failure of the script without this, except in ipython: