Fresh water is only to be pumped from the ground. The surface water is contaminated, but can be cleaned.
Limit the Power generation with steam, makes the early game harder.

Inspired by https://mods.factorio.com/mod/FiniteWater
Credits go to DellAquila