Added some database stuff to test