Started cleaning update_viz