diff --git a/README.md b/README.md
index 2eef3d1687c913c69389fab9e585e39c69db652c..26b049c34d95ee5a66529a70c77d4da23310f64a 100644
--- a/README.md
+++ b/README.md
@@ -14,6 +14,10 @@ touch __init__.py
 echo -e "import sys\nfrom os.path import dirname, join\nsys.path.insert(0, join(dirname(__file__), '.'))" >> __init__.py;
 cd -
 
+# replace $KITTI_ROOT with path to kitti
+# replace * with the script you want to run
+KITTI_ROOT=$KITTI_ROOT ./scripts/run_*.sh
+
 ```
 
 
diff --git a/main_lost.py b/main_lost.py
index c8e6b0b08be8454d34a67247d964ee96aeb5fd68..967395dfc887cf934ef6844bd01bf991e2f9e06c 100755
--- a/main_lost.py
+++ b/main_lost.py
@@ -128,7 +128,7 @@ if __name__ == "__main__":
     if args.image_path is not None:
         dataset = ImageDataset(args.image_path)
     elif args.dataset == "KITTI":
-        dataset = ImageFolderDataset("/root/kitti/training/image_2/") # TODO dont hard code
+        dataset = ImageFolderDataset("$KITTI_ROOT/training/image_2/") # TODO dont hard code
     else:
         dataset = Dataset(args.dataset, args.set, args.no_hard)