Updating training script to use new sqlite3 system