add cosmetic changes in DB class