Add matplotlib dependency