fixing small bugs in UI