Add variable friction to storm surge code