Write SQL and Python for data prep