work on unit tests