diff --git a/HOGVAX/draw_hog.py b/HOGVAX/draw_hog.py
index 8327267221fad7238608ed3aa4fc7c455d0c58f4..23b238e93b81a7c0308ed33b64fe8ac792e736db 100644
--- a/HOGVAX/draw_hog.py
+++ b/HOGVAX/draw_hog.py
@@ -29,4 +29,4 @@ def draw_hog(hog, leaves, vals, key, draw_all=False):
         attr = node.get_attributes()
         node.set_label(node.get_name() + ', ' + attr['string'] + ', ' + attr['interval_size'])
 
-    dot.write_png('peptides_output/' + key + '_colored_graph.png')
\ No newline at end of file
+    dot.write_png(key)
\ No newline at end of file
diff --git a/HOGVAX/hogvax.py b/HOGVAX/hogvax.py
index 608906234c9cf6e7009b5665d5a18247997eb05e..edeb38b0a97d01e68652cfcd7bc9364b8ee9d807 100644
--- a/HOGVAX/hogvax.py
+++ b/HOGVAX/hogvax.py
@@ -60,6 +60,8 @@ def main():
     args = get_parser().parse_args()
 
     if args.outdir:
+        if '/' not in args.outdir:
+            args.outdir = args.outdir + '/'
         if not os.path.exists(args.outdir):
             os.mkdir(args.outdir)
 
@@ -69,7 +71,7 @@ def main():
     pep_count = len(peptides)
 
     drawing_enabled = False
-    if pep_count < 30:
+    if pep_count <= 30:
         print('Number of peptides below 30 -> drawing enabled')
         drawing_enabled = True