Added functionality to DB class