Added DB code
Added DB code