ENH: add matplotlib stuff