Further fixes in Baum-Welch