Fixed minor display issues