Add code for database connection; untested