Is there a possibility that this could be updated for the generated Jupyter notebook to assert against variables the given `answer(n)`?