Tweaks

Some UI tweaks after publishing to clean up