Thank you so much 🤗 The beam is a bit finicky, there are some edge cases that go wrong 😅 But here is the source code, the code for the beam is over here.
