Added visualization options to testing