fixed small display bugs