Fixes in Baum-Welch