Water treatment for power stations