slight changes with ui