Updating main for data pipeline