Changed context_distribution to be more flexible and plot results.