Testing DB code is working