Adding database tests