Added DB code