prep_beam_plot function

Prepare for BEAM plotting