Added matplotlib to demo