Dev

Added marginal_probability function