Fixed some minor display issues