Basic plotting with pyplot