added tests for ui module