add visualization script