Add back cosZ