minor cleanup of UI