Fixed UI things.

It just looks and works better now.