Began work on new visualiser. shallow_water.py should be unchanged. Edit the comments at line 113-114 to enable the new visualiser.