Break up into more if statements
Matplotlib test import