Stay as lightning main, that gives you the most advantage
You can switch passives from the immune to anbu assault if you want leech

I think it's better to only go to Azure if you have Mei + Ao to go with chojuro