Assignment 1: Data prep in Python