working on unit tests...