now importing matplotlib as well