Experimenting with matplotlib