Add DB connection check