Simple plotting using matplotlib