Adding in basic DB class