add unit test to UI