improve viz

add styles
refact code