MNT: added matplotlib back