This PR aims at adding this feature…

Proposed Changes

I changed the foo function so that …