Used list comprehensions in marginal contributions
Made marginal_of_pi hidden