Added matplotlib plotting