Added correct matplotlib initialization
wip