Add visualization script