diff --git a/Makefile b/Makefile
index aba6059..2b8e2ce 100644
--- a/Makefile
+++ b/Makefile
@@ -1,10 +1,14 @@
-.PHONY: style quality install tensorboard clean fix update-worklog
+.PHONY: style quality install tensorboard clean fix update-worklog test
 
 # make sure to test the local checkout in scripts and not the pre-installed one
 export PYTHONPATH = src 
 
 check_dirs := . src notebooks scripts tests
 
+# Run tests
+test:
+	pytest -v tests/
+
 # Development dependencies
 install:
 	python -m venv venv && . venv/bin/activate && pip install --upgrade pip
diff --git a/requirements.txt b/requirements.txt
index 3888064..8c6581d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,4 +15,5 @@ python-dotenv
 loguru
 gradio
 tensorboard
-pytest
\ No newline at end of file
+pytest
+wandb
\ No newline at end of file