Small stylistic changes to be more consistent with paper presentation