Adding code for database connection.