Add doc test