Add context to EDPolicy API