UI Improvements

Update the UI for the next release