fixed small visual bugs