- Updates to the test class to utilize visualization asset class