add test for UI