Improve ui files.

This reworks some of the UI files to proper
expand/shrink as needed so it looks fine on
desktop and small form factors like mobile.