working on unit testing