Add host checking to DB class