diff --git a/main.py b/main.py
index a7d5fd5..1d59ddf 100644
--- a/main.py
+++ b/main.py
@@ -44,6 +44,7 @@ if __name__ == "__main__":
     args = parse_args()
 
     # Initialize data loader with parsed arguments
+    # named_data_loader also has grayscale extras. TODO: remove unnamed
     train_dataloader = init_data(
         batch_size=args.bs,
         shuffle=True,
diff --git a/makefile b/makefile
index ed2be99..950c38b 100644
--- a/makefile
+++ b/makefile
@@ -4,4 +4,7 @@ lint:
 	flake8 --ignore E501 .
 
 test:
-	python main.py --alpha 1 --lr 1e-3 --max_epochs 100
\ No newline at end of file
+	python main.py --alpha 1 --lr 1e-4 --max_epochs 500
+
+search:
+	python search.py