Added visualization class structure