Add visualization scripts