added more methods in DB class