PUB to upgrade major drain at AYE to deal with flooding