modify py script to load data into db