Problem with Baum-Welch update functions