fixed more display issues