Fix documentation broken after refacturing
Added test for documentation generation