Add minor display fixes