added expectation plot