fixed some in the ui