Attempting to speed up Baum-Welch