Add some matplotlib stuff.