How do I add a column to a dataset in Python