fixed some of the UI in the extension