diff --git a/.travis.yml b/.travis.yml
index 202e885..49ce63b 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -5,10 +5,11 @@ python:
 
 install:
   - pip install -r requirements.txt
+  - pip install codecov
 
 script:
   - python setup.py install
-  - pytest -p no:warnings
+  - pytest -p no:warnings #silence the internal tensorflow warnings
 
 after_success:
     - codecov
diff --git a/alfie/test_classify.py b/alfie/test_classify.py
index 3f4b1f2..54da252 100644
--- a/alfie/test_classify.py
+++ b/alfie/test_classify.py
@@ -1,7 +1,7 @@
+#TODO - add this last set of unit tests for the classify module
 
 import pytest
 
-
 def test_placeholder():
 
 	assert True
\ No newline at end of file