ive added code for tracking mutual information over time and changed figures a bit