Add context weight tuning