Simplify and add expectation