Created some methods in DB class