Adding DB tests