Added updated files for visualization