Documentation update

Fix for documentation