fixing small ui bugs)