Minor visual bug fixes