All database related code stored in load_data.py