Playing around with matplotlib