UI revamp

And functionality, too!