improving unit test capabilities