Playing around with some data visualizations using python