Small graphical bug fixes