fixing minor ui bugs