Load shedding to be controlled as water situation improves: Ali Zafar