fixed some small issues in the UI