improved plot for comparing conditional expectation