Refactored some code to proper display missing DB connection