TD UI Polish

Fixes a few things that I felt could be improved.