Pre-processing Data in Python