Add assert and invariant keywords