Made some UI touchups