Topological electronic structure of YbMg 2 Bi 2 and CaMg 2 Bi 2