Updated to use jinja2 templates