Yes its working .. head to the settings app then select notifications from there you will find "brief notification setting" open that and you will find edge lightning on the top. You cam enable it from there.
try charging your phone while powered off..because after new update phone optimize itself and all the apps in background for a day or two so the processor works continously that uses charge and maked phone lag