There are better experts to answer questions about this paper on the forums, but for the batch norm:
You can reuse the existing parts and add your own, for inspirations you might look at the conditional batch norm comments, and this one in particular 4.
Best regards
Thomas