changed importing matplotlib as an option. Still needed to test it out