Add optional context parameter for callbacks