Update f_shallow_wat.py