Custom environments not highlighted and breaking highlighting

[BeastyBlacksmith]

Prerequisites

• [X ] Put an X in this box if you have done the following:
  • Ensured the issue has not already been raised
  • Ensured the issue is caused by this package

Description

\begin and \end of custom environments gets not highlighted.
Also for some reason the highlighting of the figure environment breaks in the example of the screenshot.
Ideally I could teach the package to highlight the julia code according to language-julia inside of the environment.