Basic plotting in Python