Almost works, fix for initial variances