Initial values now include values for * also
Algo now gets marginals instead of max_marginals. Formatted code