added visual-experiments/waves_and_heat_map.py