Add Waterflood Control