minor UI fix

fixes to the ui