Pakistan took a 158-run lead