Add database connection method