Made some minor UI updates