Minor graphical improvement