fixed small ui issues