more work on unit testing