Added database and wrangle code