Add unit test capabilities