Make matplotlib optional