fixed some display issues