Add context parameter