Added matplotlib to mock