fixed small ui bugs