Fixed units bug (Travis). Added 1 unit test to tests.py