refactor viz
checkpoint