fixed small little issues