Update EquationBeamSearch.java