working on unit tests.