add matplotlib 2D plotting