now working with ui