Added Python scripts for visualization