added visualize_castalia.py