Add context option to run functions in