UI work.

The UI is almost done. It just needs a few more things.