Added some touch-ups to the UI. Still kinda ugly though.