fixed bug in get_stress function where original function was not returned. Added a docstring to explain why we do that.