Add BaumWelch code