Minor graphical improvements