Just some minor UI cleanup