Add docstring at top of engine.py mistakenly removed