Added functionality to db connector