added tests for UI