Added tests for simple_ui.py