update documentation with correct imports