fixed small issues with ui