add beam saving