Added matplotlib stuff to plot performance of agents