Enable hsnips in more math environments

[YuruiHong]

Some rather common math environments for me, such as flalign, were not included in the previous code. I hope this complete that.