Minor UI tweaks
Forgotten files