add matplotlib magic