- Some minor UI updates