Upgrades to a treatment site in East Sussex could reduce sewage being discharged into the sea during storms.