adding unit test capability