fixed some issues with the UI