Fix some aligns and the use of \exp (instead of \expectation)