Adding Unit tests.