Add plot test script for testing matplotlib