Update jinja_context.py

Allow filename argument