some updated in the viz module. Not working properly yet