Adding doc test