Add context param to DP