Fast-beam-alignment Code for the paper "Fast Beam Alignment via Pure Exploration in Multi-armed Bandits"